from baidu.hogwarts import Hogwarts
import time
from selenium import webdriver
from selenium.webdriver.common.by import By

class BaiDuMain:
    """
    封装的一些方法或说是接口
    """
    def __init__(self):
        self.driver = webdriver.Chrome()
        # 设置隐式等待时间,global(0.5s/轮巡)
        self.driver.implicitly_wait(5)
        self.driver.get("https://www.baidu.com")

    def send_keys(self, key):
        return self.driver.find_element(By.ID, "kw").send_keys(key)

    def click(self):
        pass

    def title(self):
        pass

    # 一个页面点击跳转到另一个页面，并返回这个页面
    def click_hogwarts_link(self):
        # click link
        # 索框中输入关键字"霍格沃兹测试学院"
        self.send_keys("霍格沃兹测试学院")
        # 点击搜索按钮
        self.driver.find_element(By.XPATH, '//*[@id="su"]').click()
        # 找到连接并点击”霍格沃兹测试学院“首页
        self.driver.find_element(By.LINK_TEXT, '霍格沃兹测试学院-软件自动化测试开发培训_接口性能测试').click()
        # time.sleep(3)
        return self.driver